from langchain.agents import initialize_agent, AgentType, load_tools
from langchain.llms.fake import FakeListLLM
# "PythonREPLTool", "PythonAstREPLTool"
tools = load_tools(["python_repl_ast"])
responses = ["Action: Python REPL\nAction Input: print(2+2)", "Final Answer: 4"]

llm = FakeListLLM(responses=responses)

agent = initialize_agent(tools=tools, llm=llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True)

agent.run("whats 5+5")
